import chromadb

# 基于内存的形式创建客户端
#client = chromadb.Client()

# 持久化到本地
client = chromadb.PersistentClient(path=r"D:\software\chroma")
cll = client.get_or_create_collection(name="test")

# 添加数据
cll.add(
    documents=["atyical by jogn","aritcal by jack","artical by jill"], # 文本内容列表，每个元素是一段文本（如文章、句子等）
    embeddings=[[1,2,3],[4,5,6],[7,8,9]], # 嵌入向量列表，每个元素是一个与documents 对应的向量表示
    ids=["1","2","3"] # 自定义ID 列表，用于唯一标识每一条记录
)

# 查询数据
print(cll.get(ids=['1']))
